from histogram import HistogramCollection, Histogram1D, Histogram2D

hcoll = HistogramCollection(
    [Histogram2D(name='kp_svd_nhits',
                 nbins=(13, 13),
                 aranges=((-0.5, 12.5),(-0.5,12.5)),
                 xlabel=r'$K^+ N_z\left[SVD\right]$',
                 ylabel=r'$K^+ N_{r\phi}\left[SVD\right]$',
                 xunits=r'',
                 yunits=r'',
                 assoc_cols=('kp_svd_nhits_z', 'kp_svd_nhits_rphi'),
                 ),
     Histogram2D(name='km_svd_nhits',
                 nbins=(13, 13),
                 aranges=((-0.5, 12.5),(-0.5,12.5)),
                 xlabel=r'$K^- N_z\left[SVD\right]$',
                 ylabel=r'$K^- N_{r\phi}\left[SVD\right]$',
                 xunits=r'',
                 yunits=r'',
                 assoc_cols=('km_svd_nhits_z', 'km_svd_nhits_rphi'),
                 ),
     Histogram2D(name='pip_svd_nhits',
                 nbins=(13, 13),
                 aranges=((-0.5, 12.5),(-0.5,12.5)),
                 xlabel=r'$\pi^+ N_z\left[SVD\right]$',
                 ylabel=r'$\pi^+ N_{r\phi}\left[SVD\right]$',
                 xunits=r'',
                 yunits=r'',
                 assoc_cols=('pip_svd_nhits_z', 'pip_svd_nhits_rphi'),
                 ),
     Histogram2D(name='pim_svd_nhits',
                 nbins=(13, 13),
                 aranges=((-0.5, 12.5),(-0.5,12.5)),
                 xlabel=r'$\pi^- N_z\left[SVD\right]$',
                 ylabel=r'$\pi^- N_{r\phi}\left[SVD\right]$',
                 xunits=r'',
                 yunits=r'',
                 assoc_cols=('pim_svd_nhits_z', 'pim_svd_nhits_rphi'),
                 ),
     ])
